package 单调栈;

import java.util.*;

/**
 * @PackageName: 单调栈
 * @ClassName: LeetCode636
 * @Author: chen jinxu
 * @Date: 2022/8/7 10:14
 * @Description: //函数的独占时间
 * 有一个 单线程 CPU 正在运行一个含有 n 道函数的程序。每道函数都有一个位于  0 和 n-1 之间的唯一标识符。
 * 函数调用 存储在一个 调用栈 上 ：当一个函数调用开始时，它的标识符将会推入栈中。而当一个函数调用结束时，
 * 它的标识符将会从栈中弹出。标识符位于栈顶的函数是 当前正在执行的函数 。
 * 每当一个函数开始或者结束时，将会记录一条日志，包括函数标识符、是开始还是结束、以及相应的时间戳。
 * 给你一个由日志组成的列表 logs ，其中 logs[i] 表示第 i 条日志消息，
 * 该消息是一个按 "{function_id}:{"start" | "end"}:{timestamp}"
 * 进行格式化的字符串。例如，"0:start:3" 意味着标识符为 0 的函数调用在时间戳 3 的
 * 起始开始执行 ；而 "1:end:2" 意味着标识符为 1 的函数调用在时间戳 2 的 末尾结束执行。注意，函数可以 调用多次，可能存在递归调用 。
 * 函数的 独占时间 定义是在这个函数在程序所有函数调用中执行时间的总和，
 * 调用其他函数花费的时间不算该函数的独占时间。例如，如果一个函数被调用两次，
 * 一次调用执行 2 单位时间，另一次调用执行 1 单位时间，那么该函数的 独占时间 为 2 + 1 = 3 。
 * 以数组形式返回每个函数的 独占时间 ，其中第 i 个下标对应的值表示标识符 i 的函数的独占时间。
 * <p>
 * 输入：n = 2, logs = ["0:start:0","1:start:2","1:end:5","0:end:6"]
 * 输出：[3,4]
 * 解释：
 * 函数 0 在时间戳 0 的起始开始执行，执行 2 个单位时间，于时间戳 1 的末尾结束执行。
 * 函数 1 在时间戳 2 的起始开始执行，执行 4 个单位时间，于时间戳 5 的末尾结束执行。
 * 函数 0 在时间戳 6 的开始恢复执行，执行 1 个单位时间。
 * 所以函数 0 总共执行 2 + 1 = 3 个单位时间，函数 1 总共执行 4 个单位时间。
 * <p>
 * 输入：n = 1, logs = ["0:start:0","0:start:2","0:end:5","0:start:6","0:end:6","0:end:7"]
 * 输出：[8]
 * <p>
 * 输入：n = 2, logs = ["0:start:0","0:start:2","0:end:5","1:start:6","1:end:6","0:end:7"]
 * 输出：[7,1]
 * 输入：n = 2, logs = ["0:start:0","0:start:2","0:end:5","1:start:7","1:end:7","0:end:8"]
 * 1 <= n <= 100
 * 1 <= logs.length <= 500
 * 0 <= function_id < n
 * 0 <= timestamp <= 109
 * 两个开始事件不会在同一时间戳发生
 * 两个结束事件不会在同一时间戳发生
 * 每道函数都有一个对应 "start" 日志的 "end" 日志   
 */
public class LeetCode636 {
    public static void main(String[] args) {
        List<String> logs = Arrays.asList(new String[]{"0:start:0", "0:start:2", "0:end:5", "0:start:6", "0:end:6", "0:end:7"});
        System.out.println(Arrays.toString(exclusiveTime(1, logs)));
        List<String> logs1 = Arrays.asList(new String[]{"0:start:0", "0:start:2", "0:end:5", "1:start:6", "1:end:6", "0:end:7"});
        System.out.println(Arrays.toString(exclusiveTime(2, logs1)));
        List<String> logs2 = Arrays.asList(new String[]{"0:start:0", "0:start:2", "0:end:5", "1:start:7", "1:end:7", "0:end:8"});
        System.out.println(Arrays.toString(exclusiveTime(2, logs2)));

    }

    public static int[] exclusiveTime(int n, List<String> logs) {
        Deque<int[]> stack = new ArrayDeque<>();
        int[] res = new int[n];
        for (String log : logs) {
            int idx = Integer.parseInt(log.substring(0, log.indexOf(":")));
            String type = log.substring(log.indexOf(":") + 1, log.lastIndexOf(":"));
            int timeTemp = Integer.parseInt(log.substring(log.lastIndexOf(":") + 1));
            if ("start".equals(type)) {
                if (!stack.isEmpty()) {
                    res[stack.peek()[0]] += timeTemp - stack.peek()[1];
                }
                stack.push(new int[]{idx, timeTemp});
            } else {
                int[] t = stack.pop();
                res[t[0]] += timeTemp - t[1] + 1;
                if (!stack.isEmpty()) {
                    stack.peek()[1] = timeTemp + 1;
                }
            }
        }
        return res;
    }
}
